#!/bin/bash

# List of languages to train on
languages=("racket" "lua" "julia" "r")
MODEL_NAME="codellama/CodeLlama-7b-hf"

# Loop over each language and run the training script
for lang in "${languages[@]}"; do
    echo "Training model for language: $lang"
    accelerate launch train/train.py --lang "$lang" --model-name "$MODEL_NAME"
done

